算子库支持情况
MindSpore Lite支持不同硬件后端的算子列表
算子名称 |
CPU |
FT04 |
FT78 |
PNNA |
|---|---|---|---|---|
Abs |
FP16 |
FP32 |
||
AbsGrad |
FP32 |
|||
Activation |
FP16 |
FP32 |
||
ActivationGrad |
FP16 |
|||
Adam |
FP32 |
|||
AddFusion |
FP16 |
FP16 |
FP32 |
FP32 |
AdderFusion |
FP32 |
|||
AddGrad |
FP32 |
|||
AddN |
FP16 |
FP32 |
||
Affine |
FP32 |
|||
All |
FP32 |
|||
AllGather |
FP32 |
|||
ApplyMomentum |
FP32 |
FP16 |
FP32 |
|
Assert |
FP16 |
Int32 |
Int32 |
|
Assign |
FP32 |
|||
ArgmaxFusion |
FP16 |
FP32 |
||
ArgminFusion |
FP16 |
FP32 |
||
AvgPoolFusion |
FP16 |
FP32 |
||
Attention |
- |
FP32 |
- |
|
AvgPoolGrad |
FP16 |
|||
BatchNorm |
FP16 |
|||
BatchNormGrad |
FP16 |
|||
BatchToSpace |
FP32 |
FP32 |
||
BatchToSpaceND |
FP16 |
|||
BiasAdd |
FP16 |
|||
BiasAddGrad |
FP16 |
|||
BinaryCrossEntropy |
FP32 |
|||
BinaryCrossEntropyGrad |
FP32 |
|||
BroadcastTo |
FP16 |
|||
Call |
FP16 |
|||
Cast |
FP16 |
|||
Ceil |
FP16 |
FP32 |
||
Clip |
FP32 |
FP32 |
||
Concat |
FP16 |
FP32 |
||
ConstantOfShape |
FP16 |
|||
Conv2DFusion |
FP16 |
FP32 |
||
Conv2DBackpropFilterFusion |
FP16 |
|||
Conv2DBackpropInputFusion |
FP16 |
|||
Conv2dTransposeFusion |
FP16 |
FP32 |
||
Cos |
FP16 |
|||
Crop |
FP16 |
|||
CropAndResize |
FP32 |
|||
CumSum |
FP32 |
|||
CustomExtractFeatures |
FP32 |
|||
CustomNormalize |
FP32 |
|||
CustomPredict |
FP32 |
|||
DEConv2DGradFilter |
FP32 |
|||
DepthToSpace |
FP16 |
FP32 |
||
DetectionPostProcess |
FP32 |
|||
DivFusion |
FP16 |
FP16 |
FP32 |
FP32 |
DivGrad |
FP32 |
|||
Dropout |
FP16 |
|||
DropoutGrad |
FP16 |
|||
DynamicQuant |
FP32 |
|||
Eltwise |
FP16 |
|||
Elu |
FP16 |
|||
Equal |
FP16 |
|||
EmbeddingLookupFusion |
FP32 |
|||
Erf |
FP16 |
|||
ExpFusion |
FP16 |
FP16 |
FP32 |
|
ExpandDims |
FP16 |
|||
Fill |
FP16 |
|||
Flatten |
FP16 |
|||
FlattenGrad |
FP16 |
|||
Floor |
FP16 |
|||
FloorDiv |
FP16 |
|||
FloorMod |
FP16 |
|||
FullConnection |
FP16 |
|||
FusedBatchNorm |
FP16 |
|||
GatherNd |
FP16 |
|||
Gather |
FP16 |
|||
GatherD |
FP16 |
|||
GLU |
FP32 |
|||
Greater |
FP16 |
|||
GreaterEqual |
FP16 |
|||
GroupNormFusion |
FP32 |
|||
GRU |
FP16 |
|||
HashtableLookup |
FP32 |
|||
InstanceNorm |
FP16 |
|||
InvertPermutation |
FP16 |
|||
IsFinite |
FP32 |
|||
L2NormalizeFusion |
FP32 |
|||
LayerNormFusion |
FP16 |
|||
LayerNormGrad |
FP16 |
|||
LeakyReLU |
FP16 |
|||
Less |
FP16 |
|||
LessEqual |
FP16 |
|||
LinSpace |
- |
FP16 |
FP32 |
- |
LRN |
FP32 |
|||
Log |
FP16 |
|||
Log1p |
FP32 |
|||
LogGrad |
FP16 |
|||
LogicalAnd |
FP16 |
|||
LogicalNot |
FP16 |
|||
LogicalOr |
FP16 |
|||
LogSoftmax |
FP16 |
|||
LshProjection |
FP32 |
|||
LSTM |
FP16 |
|||
LSTMGrad |
FP32 |
|||
LSTMGradData |
FP32 |
|||
LSTMGradWeight |
FP32 |
|||
MatMulFusion |
FP16 |
FP16 |
FP32 |
|
Maximum |
FP16 |
|||
MaximumGrad |
FP16 |
|||
MaxPoolFusion |
FP16 |
FP32 |
||
MaxPoolGrad |
FP16 |
|||
Merge |
FP16 |
|||
Minimum |
FP16 |
|||
MinimumGrad |
FP16 |
|||
Mod |
FP32 |
|||
MulFusion |
FP16 |
FP16 |
FP32 |
FP32 |
MulGrad |
FP32 |
|||
Neg |
FP16 |
FP32 |
||
NegGrad |
FP16 |
|||
NLLLoss |
FP32 |
|||
NLLLossGrad |
FP32 |
|||
NotEqual |
FP16 |
|||
NonMaxSuppression |
FP32 |
|||
NonZero |
Bool |
|||
OneHot |
FP16 |
FP32 |
||
OnesLike |
FP16 |
|||
PadFusion |
FP16 |
FP32 |
||
PartialFusion |
FP16 |
|||
PowFusion |
FP16 |
|||
PowerGrad |
FP32 |
|||
PriorBox |
FP32 |
|||
PReLUFusion |
FP16 |
|||
QuantDTypeCast |
FP16 |
FP32 |
||
RaggedRange |
FP16 |
FP16 |
FP32 |
- |
RandomNormal |
FP16 |
|||
RandomStandardNormal |
FP16 |
|||
Range |
FP16 |
FP16 |
FP32 |
- |
Rank |
FP16 |
|||
RealDiv |
FP16 |
|||
Reciprocal |
FP16 |
|||
ReduceFusion |
FP16 |
|||
ReduceScatter |
FP32 |
|||
Reshape |
FP16 |
|||
Resize |
FP16 |
|||
ResizeGrad |
FP16 |
|||
ReverseV2 |
FP32 |
|||
ReverseSequence |
FP32 |
|||
ROIPooling |
FP32 |
|||
Round |
FP16 |
|||
Rsqrt |
FP16 |
|||
RsqrtGrad |
FP32 |
|||
Select |
FP32 |
|||
Selu |
- |
|||
ScaleFusion |
FP16 |
|||
ScatterNd |
FP16 |
|||
ScatterNdUpdate |
FP16 |
|||
SGD |
FP32 |
|||
Shape |
FP16 |
|||
SigmoidCrossEntropyWithLogits |
FP32 |
|||
SigmoidCrossEntropyWithLogitsGrad |
FP32 |
|||
Sin |
FP16 |
|||
Size |
FP16 |
|||
SliceFusion |
FP16 |
|||
SkipGram |
FP32 |
|||
SmoothL1Loss |
FP32 |
|||
SmoothL1LossGrad |
FP32 |
|||
Softmax |
FP16 |
FP32 |
||
SoftmaxGrad |
FP32 |
|||
Softplus |
FP16 |
|||
SpaceToBatch |
FP16 |
|||
SpaceToBatchND |
FP16 |
|||
SpaceToDepth |
FP16 |
|||
SparseToDense |
FP16 |
|||
SparseSoftmaxCrossEntropyWithLogits |
FP32 |
|||
Splice |
FP16 |
|||
Split |
FP16 |
FP32 |
||
SplitWithOverlap |
FP16 |
|||
Sqrt |
FP16 |
|||
SqrtGrad |
FP32 |
|||
Square |
FP16 |
|||
SquaredDifference |
FP16 |
|||
Squeeze |
FP16 |
|||
StridedSlice |
FP16 |
|||
StridedSliceGrad |
FP16 |
|||
Stack |
FP16 |
|||
SubFusion |
FP16 |
FP16 |
FP32 |
FP32 |
SubGrad |
FP32 |
|||
Switch |
FP16 |
|||
SwitchLayer |
FP16 |
|||
TensorListFromTensor |
FP16 |
|||
TensorListGetItem |
FP16 |
|||
TensorListReserve |
FP16 |
|||
TensorListSetItem |
FP16 |
|||
TensorListStack |
FP16 |
|||
TensorScatterAdd |
FP32 |
|||
TileFusion |
FP16 |
|||
TopKFusion |
FP16 |
|||
Transpose |
FP16 |
FP32 |
||
UniformReal |
FP32 |
|||
Unique |
FP16 |
|||
UnsortedSegmentSum |
FP16 |
|||
Unsqueeze |
FP16 |
FP32 |
||
Unstack |
FP16 |
|||
Where |
FP16 |
|||
ZerosLike |
FP16 |
MindSpore Lite支持不同硬件后端的自定义算子列表:
算子名称 |
FT04 |
FT78 |
|---|---|---|
IFFT |
Cplx64 |
Cplx64 |
FFT |
Cplx64 |
Cplx64 |